package com.chis.encode.aes;

import com.alibaba.fastjson.JSON;
import com.chis.jm.pojo.Child;

import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import javax.crypto.SecretKey;
import javax.crypto.spec.SecretKeySpec;
import java.security.SecureRandom;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

public class AesUtil {

    public static byte[] encrypt(byte[] content, String password) {
        try {
            KeyGenerator kgen = KeyGenerator.getInstance("AES");// 创建AES的Key生产者
            kgen.init(128, new SecureRandom(password.getBytes()));// 利用用户密码作为随机数初始化出
            byte[] enCodeFormat = kgen.generateKey().getEncoded();// 返回基本编码格式的密钥，如果此密钥不支持编码，则返回
//            Cipher cipher = Cipher.getInstance("AES");// 创建密码器
            Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");//"算法/模式/补码方式"
            cipher.init(Cipher.ENCRYPT_MODE, new SecretKeySpec(enCodeFormat, "AES"));// 初始化为加密模式的密码器
            return cipher.doFinal(content);// 加密
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    public static byte[] encrypt(String content, String password) {
        try {
            KeyGenerator kgen = KeyGenerator.getInstance("AES");// 创建AES的Key生产者
            kgen.init(128, new SecureRandom(password.getBytes()));// 利用用户密码作为随机数初始化出
            // 128位的key生产者
            //加密没关系，SecureRandom是生成安全随机数序列，password.getBytes()是种子，只要种子相同，序列就一样，所以解密只要有password就行
//            SecretKey secretKey = kgen.generateKey();// 根据用户密码，生成一个密钥
            byte[] enCodeFormat = kgen.generateKey().getEncoded();// 返回基本编码格式的密钥，如果此密钥不支持编码，则返回
//            SecretKeySpec key = new SecretKeySpec(enCodeFormat, "AES");// 转换为AES专用密钥
            Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");// 创建密码器
//            byte[] byteContent = ;
            cipher.init(Cipher.ENCRYPT_MODE, new SecretKeySpec(enCodeFormat, "AES"));// 初始化为加密模式的密码器
            return cipher.doFinal(content.getBytes("utf-8"));// 加密
        } catch (Exception e) {
            e.printStackTrace();
        } 
        return null;
    }

    public static byte[] decrypt(byte[] content, String password) {
        try {
            KeyGenerator kgen = KeyGenerator.getInstance("AES");// 创建AES的Key生产者
            kgen.init(128, new SecureRandom(password.getBytes()));
            SecretKey secretKey = kgen.generateKey();// 根据用户密码，生成一个密钥
            byte[] enCodeFormat = secretKey.getEncoded();// 返回基本编码格式的密钥
            SecretKeySpec key = new SecretKeySpec(enCodeFormat, "AES");// 转换为AES专用密钥
            Cipher cipher = Cipher.getInstance("AES");// 创建密码器
            cipher.init(Cipher.DECRYPT_MODE, key);// 初始化为解密模式的密码器
            byte[] result = cipher.doFinal(content);
            return result; // 明文

        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    public static void main(String[] args) {
        List<Child> list = new ArrayList<>();
        for(int i = 0 ; i < 1000 ; i++){
            Child child = new Child();
            child.setCreateDate(new Date());
            child.setChilAccountCode(1234);
            child.setChilAddress("江苏省无锡市新吴区新安花苑二区1023");
            child.setChilAddressdistrict("dfffffff346cvbcgfbngfffdddddddd");
            child.setChilArtimove(123423);
            child.setChilBirthday(new Date());
            child.setChilBirthno("345646464646464646");
            child.setChilCardno("3423542354534534563452345546464646464646");
            child.setChilCode(235434545454L);
            child.setChilAparetel("dsgdsfdsfdf");
            child.setChilCreatesiteName("dfdfdsdgfdsfdsfdsfds");
            child.setChilIfwxin(1);
            list.add(child);
        }
        String content = JSON.toJSONString(list);
//        String content = "美女，约吗？";
        String password = "1234";
//        System.out.println("加密之前：" + content);

//        long l1 = System.currentTimeMillis();
        // 加密
        byte[] encrypt = AesUtil.encrypt(content, password);
//        System.out.println("加密后的内容：" + new String(encrypt));

        //如果想要加密内容不显示乱码，可以先将密文转换为16进制
//        String hexStrResult = parseByte2HexStr(encrypt);

        //如果的到的是16进制密文，别忘了先转为2进制再解密
//        byte[] twoStrResult = parseHexStr2Byte(hexStrResult);

        // 解密
        byte[] decrypt = AesUtil.decrypt(encrypt, password);
        System.out.println("解密之后：" + new String(decrypt));
//        System.out.println("time:"+(System.currentTimeMillis()-l1));

    }

    /**将二进制转换成16进制
     * @param buf
     * @return
     */
    public static String parseByte2HexStr(byte buf[]) {
        StringBuffer sb = new StringBuffer();
        for (int i = 0; i < buf.length; i++) {
            String hex = Integer.toHexString(buf[i] & 0xFF);
            if (hex.length() == 1) {
                hex = '0' + hex;
            }
            sb.append(hex.toUpperCase());
        }
        return sb.toString();
    }

    /**将16进制转换为二进制
     * @param hexStr
     * @return
     */
    public static byte[] parseHexStr2Byte(String hexStr) {
        if (hexStr.length() < 1)
            return null;
        byte[] result = new byte[hexStr.length()/2];
        for (int i = 0;i< hexStr.length()/2; i++) {
            int high = Integer.parseInt(hexStr.substring(i*2, i*2+1), 16);
            int low = Integer.parseInt(hexStr.substring(i*2+1, i*2+2), 16);
            result[i] = (byte) (high * 16 + low);
        }
        return result;
    }

}
